package com.blacktry.entity.easypoidata;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.metadata.BaseRowModel;
import com.alibaba.fastjson.JSON;
import com.blacktry.service.DeriveExcelService;
import com.blacktry.service.impl.DeriveExcelServiceImpl;
import org.apache.pdfbox.jbig2.util.log.Logger;
import org.apache.pdfbox.jbig2.util.log.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

// 有个很重要的点 DemoDataListener 不能被spring管理，要每次读取excel都要new,然后里面用到spring可以构造方法传进去
public class DemoDataListener<T extends BaseRowModel> extends AnalysisEventListener<DeriveExcel> {

    private static final Logger LOGGER = LoggerFactory.getLogger(DemoDataListener.class);
    /**
     * 每隔5条存储数据库，实际使用中可以3000条，然后清理list ，方便内存回收
     */
    private static final int BATCH_COUNT = 20;
    List<DeriveExcel> list = new ArrayList<DeriveExcel>();

    /**
     * 这个每一条数据解析都会来调用
     *
     * @param data
     *            one row value. Is is same as {@link AnalysisContext#readRowHolder()}
     * @param context
     */
    @Override
    public void invoke(DeriveExcel data, AnalysisContext context) {
        System.out.println("解析到一条数据->"+data);
        list.add(data);
//         达到BATCH_COUNT了，需要去存储一次数据库，防止数据几万条数据在内存，容易OOM
        if (list.size() >= BATCH_COUNT) {
            saveData(list);
            // 存储完成清理 list
            list.clear();
        }
    }
    /**
     * 所有数据解析完成了 都会来调用
     *
     * @param context
     */
    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        // 这里也要保存数据，确保最后遗留的数据也存储到数据库
        saveData(list);
        LOGGER.info("所有数据解析完成！");
    }
    /**
     * 加上存储数据库
     */
    private List<DeriveExcel> dataList;
    private void saveData(List<DeriveExcel> list) {
        LOGGER.info("开始存储数据库！");
        DeriveExcelServiceImpl deriveExcelService = new DeriveExcelServiceImpl();
        deriveExcelService.saveOrUpdateBatch(list);
        LOGGER.info("存储数据库成功！");
    }

}